package com.campus.counseling.config;

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

@Data
@Component
@ConfigurationProperties(prefix = "lstm")
public class LSTMConfig {
    
    // 模型参数
    private int vectorSize = 200;  // Word2Vec 向量维度
    private int inputSize = 205;   // 200维词向量 + 5维情感特征
    private int hiddenSize = 128;
    private int numLayers = 2;
    private int outputSize = 1;
    private int batchSize = 32;
    private int numEpochs = 100;
    private double learningRate = 0.001;
    private long seed = 12345L;
    
    // 序列参数
    private int maxSeqLength;
    private int vocabSize;
    
    // 模型文件路径
    private String modelPath;
    private String word2vecPath;
    
    // 需要添加的 Word2Vec 配置
    private int iterations = 10;
    private int windowSize = 5;
    private int minWordFrequency = 1;
} 